from process import extract_audio_features
import argparse

if __name__ == '__main__':
  parser = argparse.ArgumentParser()
  parser.add_argument("--path", type=str, help="wav file path")
  parser.add_argument("--ckp", default='scene/checkpoints/audio_visual_encoder.pth', type=str, help="checkpoint path")
  args = parser.parse_args()
  extract_audio_features(args.path, checkpoint_path=args.ckp)